package com.zerdoor.filter;

/**
 * Created by xiaobo on 2014/9/1.
 */

import javax.servlet.*;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import javax.servlet.http.HttpSession;

import com.sun.org.apache.bcel.internal.classfile.Visitor;

import java.util.List;
import java.util.ArrayList;
import java.util.StringTokenizer;
import java.io.IOException;

/**
 * 用于检测用户是否登陆的过滤器，如果未登录，则重定向到指的登录页面
 * <p/>
 * 配置参数
 * <p/>
 * checkSessionKey 需检查的在 Session 中保存的关键字<br/>
 * redirectURL 如果用户未登录，则重定向到指定的页面，URL不包括 ContextPath<br/>
 * notCheckURLList 不做检查的URL列表，以分号分开，并且 URL 中不包括 ContextPath<br/>
 */
public class CheckLoginFilter implements Filter {
    protected FilterConfig filterConfig = null;
    private String redirectURL = null;
    private List notCheckURLList = new ArrayList();
    private List checkURLList = new ArrayList();
    private String sessionKey = null;

    public void doFilter(ServletRequest servletRequest,
                         ServletResponse servletResponse, FilterChain filterChain)
            throws IOException, ServletException {
        servletRequest.setCharacterEncoding("UTF-8");
        servletResponse.setContentType("text/html;charset=UTF-8");
        servletResponse.setCharacterEncoding("UTF-8");

        HttpServletRequest request = (HttpServletRequest) servletRequest;
        HttpServletResponse response = (HttpServletResponse) servletResponse;

        HttpSession session = request.getSession();

        if (sessionKey == null) {
            filterChain.doFilter(request, response);
            return;
        }
        if ((checkRequestURIIntFilterList(request))
                && session.getAttribute(sessionKey) == null) {


//			if(request.getRequestURI().contains("index")){
//				response.sendRedirect(request.getContextPath() + redirectURL);
//			}else{
            HttpServletRequest req = (HttpServletRequest) request;
            Visitor visitor = (Visitor) req.getSession().getAttribute("visitor");
            if (visitor == null) {
                java.io.PrintWriter out = response.getWriter();
                out.println("<html>");
                out.println("<script>");
                out.println("window.top.location='" + request.getContextPath() + "/login/index.html';");
                out.println("</script>");
                out.println("</html>");
            }
//			}
            return;
        }
        filterChain.doFilter(servletRequest, servletResponse);
    }

    public void destroy() {
        notCheckURLList.clear();
    }

    private boolean checkRequestURIIntNotFilterList(HttpServletRequest request) {
        String uri = request.getServletPath()
                + (request.getPathInfo() == null ? "" : request.getPathInfo());
        boolean re = false;
        for (Object o : notCheckURLList) {
            if (uri.contains(String.valueOf(o))) {
                re = true;
                break;
            }
        }
        return re;
//		return notCheckURLList.contains(uri);
    }

    private boolean checkRequestURIIntFilterList(HttpServletRequest request) {
        String uri = request.getServletPath()
                + (request.getPathInfo() == null ? "" : request.getPathInfo());
        boolean re = false;
        for (Object o : checkURLList) {
            if (uri.contains(String.valueOf(o))) {
                re = true;
                break;
            }
        }
        return re;
//		return notCheckURLList.contains(uri);
    }

    public void init(FilterConfig filterConfig) throws ServletException {
        this.filterConfig = filterConfig;
        redirectURL = filterConfig.getInitParameter("redirectURL");
        sessionKey = filterConfig.getInitParameter("checkSessionKey");

        String notCheckURLListStr = filterConfig
                .getInitParameter("notCheckURLList");

        if (notCheckURLListStr != null) {
            StringTokenizer st = new StringTokenizer(notCheckURLListStr, ";");
            notCheckURLList.clear();
            while (st.hasMoreTokens()) {
                String s = st.nextToken();
                notCheckURLList.add(s);
            }
        }

        String checkURLListStr = filterConfig
                .getInitParameter("checkURLList");

        if (checkURLListStr != null) {
            StringTokenizer st2 = new StringTokenizer(checkURLListStr, ";");
            checkURLList.clear();
            while (st2.hasMoreTokens()) {
                String s = st2.nextToken();
                checkURLList.add(s);
            }
        }
    }

}
